import torch
from torchline.losses import LOSS_FN_REGISTRY

__all__ = [
    'MyLoss'
]
@LOSS_FN_REGISTRY.register()
def MyLoss(cfg):
    return torch.nn.MSELoss()
